import os
import openai

# gpt 3.5
openai.organization = "org-loTDRUIJ0COsl0FpmN89NrIK"
openai.api_key ="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"
print(openai.Model.list())

def generate_prompt(animal):
    return """Suggest three names for an animal that is a superhero.

Animal: Cat
Names: Captain Sharpclaw, Agent Fluffball, The Incredible Feline
Animal: Dog
Names: Ruff the Protector, Wonder Canine, Sir Barks-a-Lot
Animal: {}
Names:""".format(animal.capitalize())

animal = "Horse"

response = openai.Completion.create(
  model="text-davinci-003",
  prompt=generate_prompt(animal),
  temperature=0.6
)